"""
演示第三个图标，GDP动态柱状图开发
"""
from pyecharts.charts import Bar,Timeline
from pyecharts.globals import ThemeType
from pyecharts.options import LabelOpts, TitleOpts

# 读取数据
f = open("1960-2019全球GDP数据.txt","r",encoding="GB2312")
data_line = f.readlines()
# 关闭文件
f.close()
# 删除第一条数据
data_line.pop(0)
# 将数据转换成字典存储，格式为：
# { 年份：[[国家,gdp],[国家,gdp],......],年份：[[国家,gdp],[国家，gdp],......}
# 先定义一个字典
data_dict = {}
for line in data_line:
    year = int(line.split(",")[0])  # 年份
    country = line.split(",")[1]    # 国家
    gdp = float(line.split(",")[2])  # gdp数据
    # 如何判断字典里有没有指定的key呢？
    try:
        data_dict[year].append([country,gdp])
    except KeyError:
        data_dict[year] = []
        data_dict[year].append([country, gdp])
# 创建时间线对象
timeline = Timeline({"theme":ThemeType.LIGHT})
# 排序年份
sorted_year_list = sorted(data_dict.keys())
for year in sorted_year_list:
    data_dict[year].sort(key=lambda x:x[1],reverse=True)
    # 取出本年份前8名的国家
    year_data = data_dict[year][0:8]
    x_data = []
    y_data = []
    for country_gdp in year_data:
        x_data.append(country_gdp[0]) # x轴添加国家数据
        y_data.append(country_gdp[1]/100000000) # y轴添加gdp数据

    # 构建柱状图
    bar = Bar()
    x_data.reverse()
    y_data.reverse()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP(亿)",y_data,label_opts=LabelOpts(position="right"))
    # 设置标题
    bar.set_global_opts(
        title_opts=TitleOpts(title=f"{year}年GDP世界前八国家")
    )
    # 反转x轴和y轴
    bar.reversal_axis()
    # 添加时间线数据
    timeline.add(bar,str(year))

    # 设置时间线自动播放
    timeline.add_schema(
        play_interval=1000,
        is_timeline_show=True,
        is_auto_play=True,
        is_loop_play=True,
    )
# 绘图
timeline.render("1960~2019GDP前8国家.html")